hackerrank day 1 data types problem solution in python